package orm

import (
	"base-gin/app/config"
	"fmt"
	"go.uber.org/zap"
	"gorm.io/driver/mysql"
	"gorm.io/gorm"
	"moul.io/zapgorm2"
	"time"
)

var db *gorm.DB

func init() {
	dsn := fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?charset=utf8mb4&parseTime=True&loc=Local",
		config.GormConfig.User, config.GormConfig.Password, config.GormConfig.Host, config.GormConfig.Port, config.GormConfig.DB)
	var err error

	db, err = gorm.Open(mysql.Open(dsn), &gorm.Config{
		Logger: zapgorm2.New(zap.L()),
	})
	if err != nil {
		panic(fmt.Errorf("数据库连接失败%s", err))
	}
	sqlDB, err := db.DB()
	if err != nil {
		panic(fmt.Errorf("获取数据库通用接口失败%s", err))
	}
	// SetMaxIdleConns 用于设置连接池中空闲连接的最大数量。
	sqlDB.SetMaxIdleConns(config.GormConfig.MaxIdleConns)
	// SetMaxOpenConns 设置打开数据库连接的最大数量。
	sqlDB.SetMaxOpenConns(config.GormConfig.MaxOpenConns)
	// SetConnMaxLifetime 设置了连接可复用的最大时间。
	sqlDB.SetConnMaxLifetime(time.Hour)
}

func GetDb() *gorm.DB {
	return db
}
